# mongodb
from datetime import datetime

from pymongo import MongoClient
import uuid

client = MongoClient(host='59.110.227.150')
client.super_ranking.authenticate('ranking_rw', 'ranking123890')
db = client.super_ranking

categories = ['国内', '国际', '娱乐', '体育', '财经', '科技', '社会', '时尚', '汽车', '房产', '游戏', '旅游', '军事']
category_tables = {
    '国内': {'国内', '国内新闻'},
    '国际': {'国际', '国际新闻'},
    '娱乐': {'娱乐', '娱乐新闻', '娱乐排行'},
    '体育': {'体育', '体育频道', '体育排行', '体育新闻'},
    '财经': {'财经', '财经新闻', '财经排行'},
    '科技': {'科技', 'ＩＴ', '科技新闻', '互联网'},
    '社会': {'社会', '社会新闻'},
    '时尚': {'时尚'},
    '汽车': {'汽车'},
    '房产': {'房产'},
    '游戏': {'游戏'},
    '旅游': {'旅游'},
    '军事': {'军事', '军事新闻', '军事排行'},
}

website_infos = ['腾讯网', '凤凰网', '人民网', '网易新闻', '新浪新闻', '今日头条', '百度新闻', '搜狐新闻', '中国新闻', '央视网', '滂湃新闻', '联合新闻网']


def create_categories():
    for category in categories:
        db.category_info.insert(
            {'_id': str(uuid.uuid1()), 'name': category,
             'create_time': datetime.now().strftime('%Y-%m-%d %H:%M:%S')})


def create_website_infos():
    for website_info in website_infos:
        db.website_info.insert(
            {'_id': str(uuid.uuid1()), 'name': website_info,
             'create_time': datetime.now().strftime('%Y-%m-%d %H:%M:%S')})


if __name__ == '__main__':
    # create_website_infos()
    # create_categories()
    pass
